import akshare as ak
import pandas as pd
import matplotlib.pyplot as plt

# 设置图片清晰度
plt.rcParams['figure.dpi'] = 300

# 获取天通股份历史行情数据，股票代码为 600330.SH
stock_hq_df = ak.stock_zh_a_hist(symbol="600330", period="daily", start_date="20240101", end_date="20250210")

# 打印前几行数据查看
print("天通股份历史行情数据：")
print(stock_hq_df.head())

# 保存数据为 CSV 文件，设置编码为 UTF-8
csv_file_path = 'tiantong_stock_data.csv'
stock_hq_df.to_csv(csv_file_path, encoding='utf-8', index=False)
print(f"数据已保存为 {csv_file_path}")

# 保存数据为 Excel 文件，Excel 默认以 UTF-8 兼容格式保存
excel_file_path = 'tiantong_stock_data.xlsx'
stock_hq_df.to_excel(excel_file_path, index=False)
print(f"数据已保存为 {excel_file_path}")

# 绘制收盘价折线图
plt.figure(figsize=(12, 6))
plt.plot(stock_hq_df['日期'], stock_hq_df['收盘'], marker='o', linestyle='-', color='b', label='收盘价')

# 添加标题和标签
#以udf-8格
plt.rcParams['font.sans-serif'] = ['SimHei']
plt.rcParams['axes.unicode_minus'] = False
plt.title('天通股份收盘价走势（2024年1月1日 - 2025年2月10日）')
plt.xlabel('日期')
plt.ylabel('收盘价（元）')

# 旋转 x 轴标签以便更好显示
plt.xticks(rotation=45)

# 显示图例
plt.legend()

# 显示网格
plt.grid(True)

# 显示图表
plt.tight_layout()
plt.show()